﻿using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using Tracy.DataModel.UltraExplorer;
using Tracy.DataModel;

namespace Tracy.Plugins.Classifiers.EntryClassifiers
{
    public class UnwatchedEntryClassifier : AbstractClassifier<Entry>
    {
        protected override List<string> GetCategories(Entry entry)
        {
            List<string> rtn = new List<string>();
            int unWatchedCount = TracyFacade.Instance.MediaFileProvider.GetUnwatchedCount(entry);
            if (unWatchedCount > 0) rtn.Add("Unwatched");
            else rtn.Add("Watched");

            return rtn;
        }
    }
}
